package com.Questions.hash;


import java.util.*;

/**
 * 和为K的子数组
 */
public class leetcode560 {
    public int subarraySum1(int[] nums, int k) {
        // 前缀和
        int[] S=new int[nums.length+1];
        S[0]=0;
        for(int i=0;i<nums.length;++i){
            if(i==0){
                S[i+1]=nums[0];
            }else{
                S[i+1]=S[i]+nums[i];
            }
        }
        int res=0;
        for(int i=0;i<nums.length;++i){
            for(int j=i+1;j<=nums.length;++j){
                if(S[j]-S[i]==k){
                    res++;
                }
            }
        }
        return res;
    }
    public int subarraySum(int[] nums, int k) {
        int count = 0, pre = 0;
        HashMap < Integer, Integer > mp = new HashMap < > ();
        mp.put(0, 1);
        for (int i = 0; i < nums.length; i++) {
            // 只需要使用一次 pre的求和 问题就可以解决
            pre += nums[i];
            if (mp.containsKey(pre - k)) {
                count += mp.get(pre - k);
            }
            mp.put(pre, mp.getOrDefault(pre, 0) + 1);
        }
        return count;
    }
}
